cd ../../

gpus=$1
ids_bsz=24
exp_name="baseline_noMT_checkBN"
for ids_bsz in 0
do 
    for ood_bsz in 24 48 96 192
    do
        echo "Run experiments on <${exp_name}> with o${ood_bsz}_i${ids_bsz}"
        echo "batch_sizes: [0, 12, 12, ${ids_bsz}, ${ood_bsz}]"
        
        python sed_main.py \
            --gpus ${gpus}, \
            --exp_name "sed/${exp_name}/o${ood_bsz}_i${ids_bsz}" \
            --batch_sizes "[0, 12, 12, ${ids_bsz}, ${ood_bsz}]" \
            --const_max 0
    done
done